package com.heima.behavior.service.impl;

import com.alibaba.fastjson.JSON;
import com.heima.behavior.service.ApBehaviorEntryService;
import com.heima.behavior.service.ApCollectionBehaviorService;
import com.heima.common.constants.article.HotArticleConstants;
import com.heima.common.exception.CustomException;
import com.heima.model.behavior.dtos.CollectionBehaviorDTO;
import com.heima.model.behavior.pojos.ApBehaviorEntry;
import com.heima.model.behavior.pojos.ApCollection;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.mess.UpdateArticleMess;
import com.heima.model.threadlocal.AppThreadLocalUtils;
import com.heima.model.user.pojos.ApUser;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;

@Slf4j
@Service
public class ApCollectionBehaviorServiceImpl implements ApCollectionBehaviorService {
    @Autowired
    MongoTemplate mongoTemplate;
    @Autowired
    ApCollectionBehaviorService apCollectionBehaviorService;
    @Autowired
    ApBehaviorEntryService apBehaviorEntryService;
    @Autowired
    RabbitTemplate rabbitTemplate;
    @Override
    public ResponseResult collectBehavior(CollectionBehaviorDTO dto) {
        //校验参数 需要登录 文章id不能null  操作类型 0 或 1
//
        //根据userId查询行为实体数据
        //如果未收藏 新增收藏行为
//
        //如果是取消收藏操作  删除收藏行为


        //1. 校验参数 文章id不能为空  需要登录 不喜欢类型取值为 0 或 1
        ApUser user = AppThreadLocalUtils.getUser();
        if (user == null) {
            throw new CustomException(AppHttpCodeEnum.DATA_NOT_EXIST);
        }
        Integer userId = user.getId();
//        Integer userId = 123456;
        //2. 查询行为实体数据
        ApBehaviorEntry apBehaviorEntry = apBehaviorEntryService.findByUserIdOrEquipmentId(userId, dto.getEquipmentId());
        //3. //如果是收藏操作 判断是否已经收藏
        if (dto.getType() == 0) {
            //4. 如果不存在   添加收藏行为
            Query query = Query.query(Criteria.where("articleId").is(dto.getArticleId())
                    .and("entryId").is(apBehaviorEntry.getId()));
            ApCollection one = mongoTemplate.findOne(query, ApCollection.class);
            if (one == null) {
                one= new ApCollection();
                one.setEntryId(apBehaviorEntry.getId());
                one.setArticleId(dto.getArticleId());
                one.setType((short)0);

                mongoTemplate.insert(one);
            }
            // =======================新加代码==========================
            UpdateArticleMess mess = new UpdateArticleMess();
            mess.setType(UpdateArticleMess.UpdateArticleType.COLLECTION);
            mess.setArticleId(dto.getArticleId());
            mess.setAdd(dto.getOperation()==0?1:-1);
            rabbitTemplate.convertAndSend(HotArticleConstants.HOT_ARTICLE_SCORE_BEHAVIOR_QUEUE, JSON.toJSONString(mess));
            log.info("收藏行为 发送消息 ==>{}",JSON.toJSONString(mess));
            // =======================新加代码==========================

            return ResponseResult.okResult("收藏成功");
        }
        //5. 如果是 取消不喜欢操作  删除对应的不喜欢数据
        Query query = Query.query(Criteria.where("articleId").is(dto.getArticleId())
                .and("entryId").is(apBehaviorEntry.getId()));
        mongoTemplate.remove(query,ApCollection.class);
        // =======================新加代码==========================
        UpdateArticleMess mess = new UpdateArticleMess();
        mess.setType(UpdateArticleMess.UpdateArticleType.COLLECTION);
        mess.setArticleId(dto.getArticleId());
        mess.setAdd(dto.getOperation()==0?1:-1);
        rabbitTemplate.convertAndSend(HotArticleConstants.HOT_ARTICLE_SCORE_BEHAVIOR_QUEUE, JSON.toJSONString(mess));
        log.info("收藏行为 发送消息 ==>{}",JSON.toJSONString(mess));
        // =======================新加代码==========================
        return ResponseResult.okResult("取消收藏成功");


    }
}
